# -*- coding: utf-8 -*-
# © 2018 QYT Technology
# Authored by: Zhao Xingtao (zxt50330@gmail.com)
from marshmallow import fields

from bills import ma
from bills.debt.models import Income, Expenditure, Budget


class IncomeCreateSchema(ma.ModelSchema):
    class Meta:
        model = Income
        exclude = 'type',


class ExpenditureCreateSchema(ma.ModelSchema):
    class Meta:
        model = Expenditure
        exclude = 'type',


class BudgetCreateSchema(ma.ModelSchema):
    class Meta:
        model = Budget
        exclude = 'type',


class BudgetDetailSchema(ma.ModelSchema):
    class Meta:
        model = Budget
